Please run files named train.ipynb and train1.ipynb to repruduce reasults from paper
